package com.ai.langchain4jchat.aiservice;

import dev.langchain4j.service.MemoryId;
import dev.langchain4j.service.SystemMessage;
import dev.langchain4j.service.UserMessage;
import dev.langchain4j.service.spring.AiService;
import dev.langchain4j.service.spring.AiServiceWiringMode;
import reactor.core.publisher.Flux;

@AiService(
        wiringMode = AiServiceWiringMode.EXPLICIT, //手动装配
        chatModel = "openAiChatModel", //指定模型
        streamingChatModel = "openAiStreamingChatModel", //指定流式模型
        /*chatMemory = "chatMemory"*/ //配置会话记忆对象
        chatMemoryProvider = "chatMemoryProvider" //配置会话记忆提供者
)
public interface AssistantService {
    //用于聊天的方法
    //public String chat(String message);
    @SystemMessage(fromResource = "system.txt") //从资源文件加载系统消息
    //@UserMessage("你是来自三千州的荒，你的名字叫石昊是一位绝世天才！{{it}}") //用户消息模板
    //@UserMessage("你是来自三千州的荒，你的名字叫石昊是一位绝世天才！{{msg}}") //用户消息模板
    //public Flux<String> chat(@V("msg") String message);
    //public Flux<String> chat(String message);
    public Flux<String> chat(@MemoryId String memoryId, @UserMessage String message);
}
